#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@author:hengk
@contact: hengk@foxmail.com
@datetime:2019-11-01 18:01
"""

from models.resnet import  resnet
from models.pse_resnet import resnet50

class BackBoneFactory(object):

    @staticmethod
    def create(name,pretrain,num_class):
        if(name =="resnet50"):
            # return resnet(name,True,num_class)
            return resnet50(True)
        print("没有"+name+"网络结构")
# a= BackBoneFactory.create("resnet50")
